/******************************************************************************
 * Copyright (C) 671643387@qq.com
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License as published by the
 * Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
 * more details.
 *
 * You should have received a copy of the GNU General Public License along
 * with this program; if not, write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
 *
 *****************************************************************************/

#include "NodeCpp/Net/TcpAcceptor.h"
#include <cassert>
#include <cstring>
#include "NodeCpp/Platform.h"
#if defined(PLATFORM_LINUX)
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#endif
#include "NodeCpp/Platform.h"

namespace NodeCpp {
namespace Net {

    TcpAcceptor::TcpAcceptor(struct event_base *_Base, AcceptCallback _Callback, 
        void *_Userptr /* = NULL */)
        : Base_(_Base)
        , Listener_(NULL)
        , Callback_(_Callback)
        , UserPtr_(_Userptr)
    {

    }

    TcpAcceptor::~TcpAcceptor(void)
    {
        _Uninitialize();
    }

#if defined(PLATFORM_WINDOWS)
    bool TcpAcceptor::Open(const char *_Host, int _Port, int _Backlog)
    {
        struct sockaddr_in _Addr;
        memset(&_Addr, 0, sizeof(_Addr));
        _Addr.sin_family = AF_INET;
        _Addr.sin_addr.S_un.S_addr = inet_addr(_Host);
        _Addr.sin_port = htons(u_short(_Port));

        Listener_ = evconnlistener_new_bind(
            Base_, &TcpAcceptor::_AcceptCallback, this,
            LEV_OPT_REUSEABLE | LEV_OPT_CLOSE_ON_FREE, _Backlog,
            (struct sockaddr*)&_Addr,
            sizeof(_Addr));
        return Listener_ != NULL;
    }
#elif defined(PLATFORM_LINUX)
    bool TcpAcceptor::Open(const char *_Host, int _Port, int _Backlog)
    {
        struct sockaddr_in _Addr;
        memset(&_Addr, 0, sizeof(_Addr));
        _Addr.sin_family = AF_INET;
        _Addr.sin_addr.s_addr = inet_addr(_Host);
        _Addr.sin_port = htons(u_short(_Port));

        Listener_ = evconnlistener_new_bind(
            Base_, &TcpAcceptor::_AcceptCallback, this,
            LEV_OPT_REUSEABLE | LEV_OPT_CLOSE_ON_FREE, _Backlog,
            (const struct sockaddr*)&_Addr,
            sizeof(_Addr));
        return Listener_ != NULL;
    }
#endif

    void TcpAcceptor::Close(void)
    {
        if (Listener_) {
            evconnlistener_free(Listener_);
            Listener_ = NULL;
        }
    }

    bool TcpAcceptor::Enable(bool _Enable)
    {
        NODECPP_ASSERT(Listener_);
        if (_Enable) {
            return evconnlistener_enable(Listener_) == 0;
        }
        return evconnlistener_disable(Listener_) == 0;
    }


    void TcpAcceptor::_Uninitialize(void)
    {
        if (Listener_)
        {
            evconnlistener_free(Listener_);
            Listener_ = NULL;
        }
    }

	void TcpAcceptor::_AcceptCallback(struct evconnlistener *_Listener, evutil_socket_t _Sock, struct sockaddr *_Addr, int _Addrlen, void *_Userptr)
    {
        TcpAcceptor *_TcpAcceptor = static_cast<TcpAcceptor*>(_Userptr);
        NODECPP_ASSERT(_TcpAcceptor && _TcpAcceptor->Callback_);
        _TcpAcceptor->Callback_(_Sock, _Addr, _Addrlen, _TcpAcceptor->UserPtr_);
    }

}}
